1496B - Max and Mex - CodeForces Solution

#include<bits/stdc++.h>
using namespace std;
int a[100005];
int main()
{
	int t,n,k;
	scanf("%d",&t);
	while(t--)
	{
		
		scanf("%d%d",&n,&k);
		int mex=0,maxn=0;
		for(int i=0;i<n;i++)
		{
			scanf("%d",&a[i]);
		}
		sort(a,a+n);
		int ans;
		if(k==0)
		{
			ans=n;
		}else 
		{
			int flag=0;
			for(int i=0;i<n;i++)
			{
				if(a[i]!=i)
				{
					mex=i;
					flag=1;
					break;
				}
			}
			if(flag==0)
			{
				mex=n;
			}
			int x;
			maxn=a[n-1];
			if(mex<maxn)
			{
				if((mex+maxn)%2==1)
				{
					x=(mex+maxn)/2+1;
				}
				else 
				{
					x=(mex+maxn)/2;
				}	
				int temp=0;
				for(int i=0;i<n;i++)
				{
				    if(a[i]==x)
					{
						ans=n;
						temp=1;
						break;
					}	
				}			
				if(temp==0)
				{
					ans=n+1;
				}
			}else 
			{
				ans=n+k;
			}	
		}printf("%d\n",ans);
	}
}